
from ultralytics import YOLO

# 加载模型
#model = YOLO('yolov8n.pt')  # 预训练的 YOLOv8n 模型
model = YOLO('runs/detect/train/weights/best.pt')  # 使用自己训练出来的模型

# 图片列表
image_paths = ['images/00292.png', 'images/00293.png', 'images/00296.png', 'images/00298.png']

# 运行批量推理
#results = model(image_paths)  # 返回 Results 对象列表
model.predict(image_paths, save=True, imgsz=320, conf=0.5)

